/*
 * Copyright 2012-2014 the original author or authors.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */
package com.jqd.examples.jpa2;

import java.util.ArrayList;
import java.util.List;

import org.springframework.orm.jpa.persistenceunit.MutablePersistenceUnitInfo;
import org.springframework.orm.jpa.persistenceunit.PersistenceUnitPostProcessor;

/**
 * TODO
 * 
 * @author liaozhicheng.cn@gmail.com
 * @date 2012-9-2
 * @since 1.0
 */
public class LoadEntityPersistenceUnitPostProcessor implements PersistenceUnitPostProcessor {
	
	private static List<String> loadEntityClassNames = new ArrayList<String>();
	
	
	public static void registerLoadEntityClassNames(List<String> loadEntities) {
		loadEntityClassNames.clear();
		loadEntityClassNames.addAll(loadEntities);
	}
	
	
	public void postProcessPersistenceUnitInfo(MutablePersistenceUnitInfo pui) {
		for (String loadEntityClassName : loadEntityClassNames) {
			pui.addManagedClassName(loadEntityClassName);
		}
	}
	
}
